class Config:
    batch_size = 100
    img_size = 224
    path = "/home/one/lhbdata/pet/pet_biometric_challenge_2022/train/"
    epoch = 150
    model_name = "swin_base_patch4_window7_224"   #convit_base  ,swin_base_patch4_window7_224 ,seresnet50,seresnet101
    model_dir = '/home/one/pth/'
    results_dir = '/home/one/result/'
    num_class = 6000
    lr = 0.002
    momentum = 0.8
    printsize = 10
    val_path = "/home/one/lhbdata/pet/pet_biometric_challenge_2022/validation/"